package com.starhub.engine.milvus.biz;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import com.starhub.application.rag.biz.MilvusClientFactory;

/**
 * Milvus客户端工厂测试类
 * 用于验证连接池管理功能
 * 
 * @version: V1.0
 * @author: liuhf
 */
@Slf4j
@Component
public class MilvusClientFactoryTest {

    @Autowired
    private MilvusClientFactory milvusClientFactory;

    /**
     * 测试连接池管理功能
     */
    public void testConnectionPool() {
        try {
            log.info("开始测试Milvus连接池管理功能...");
            
            // 测试获取客户端
            // 注意：这里需要确保有有效的Milvus配置
            // milvusClientFactory.getClient("test_mark");
            
            // 测试清理空闲连接
            milvusClientFactory.cleanupIdleClients();
            
            log.info("Milvus连接池管理功能测试完成");
        } catch (Exception e) {
            log.error("测试Milvus连接池管理功能时发生异常", e);
        }
    }
} 